# encoding: utf-8
# author: 侯佳涛
# file_name: main
# create_time: 2021/2/26, 下午2:54
import os
import tensorflow as tf

flags = tf.app.flags
flags.DEFINE_boolean("train", False, "是否进行训练")
flags.DEFINE_boolean("dev", False, "是否进行验证")
flags.DEFINE_boolean('test', True, "是否进行测试")
flags.DEFINE_integer('batch_size', 32, '批次大小')
flags.DEFINE_integer('max_seq_len', 128, '最大句子长度')
flags.DEFINE_integer('lstm_dim', 128, 'lstm单元数量')
flags.DEFINE_string('pre_bert_path', 'chinese_L-12_H-768_A-12', 'bert预训练模型路径')
flags.DEFINE_string('bert_vocab_file', os.path.join('chinese_L-12_H-768_A-12', 'vocab.txt'), 'bert词汇文件')
flags.DEFINE_string('bert_config_file', os.path.join('chinese_L-12_H-768_A-12', 'bert_config.json'), 'bert配置文件')
flags.DEFINE_string('do_lower_case', True, 'bert_tokenizer_do_lower_case')
flags.DEFINE_string('train_data_path', os.path.join('data', 'example.train'), '训练数据路径')
flags.DEFINE_string('dev_data_path', os.path.join('data', 'example.dev'), '验证数据路径')
flags.DEFINE_string('test_data_path', os.path.join('data', 'example.test'), '测试数据路径')


